package com.hn.zhijian.base.data.common;

/**
 * @Author carl chen
 * @date 2018/2/7.
 */
public class DbContextHolder {
    private static final ThreadLocal contextHolder = new ThreadLocal<>();

//    /**
//     * 设置数据源
//     * @param dbTypeEnum
//     */
//    public static void setDbType(DBTypeEnum dbTypeEnum) {
//        contextHolder.set(dbTypeEnum.getValue());
//    }

     public static void setDbType(String value) {
        contextHolder.set(value);
    }

    /**
     * 取得当前数据源
     * @return
     */
    public static String getDbType() {
        return (String) contextHolder.get();
    }

    /**
     * 清除上下文数据
     */
    public static void clearDbType() {
        contextHolder.remove();
    }
}
